<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd">
<html xmlns="http://www.w3.org/1999/xhtml">
<head>
<meta http-equiv="Content-Type" content="text/xhtml;charset=UTF-8"/>
<meta http-equiv="X-UA-Compatible" content="IE=9"/>
<title>AMBHAS: /home/tomer/svn/ambhas/ambhas/extract_gis_data.py Source File</title>

<link href="tabs.css" rel="stylesheet" type="text/css"/>
<link href="doxygen.css" rel="stylesheet" type="text/css" />
<link href="navtree.css" rel="stylesheet" type="text/css"/>
<script type="text/javascript" src="jquery.js"></script>
<script type="text/javascript" src="resize.js"></script>
<script type="text/javascript" src="navtree.js"></script>
<script type="text/javascript">
  $(document).ready(initResizable);
</script>
<link href="search/search.css" rel="stylesheet" type="text/css"/>
<script type="text/javascript" src="search/search.js"></script>
<script type="text/javascript">
  $(document).ready(function() { searchBox.OnSelectItem(0); });
</script>

</head>
<body>
<div id="top"><!-- do not remove this div! -->


<div id="titlearea">
<table cellspacing="0" cellpadding="0">
 <tbody>
 <tr style="height: 56px;">
  
  
  <td style="padding-left: 0.5em;">
   <div id="projectname">AMBHAS
   
   </div>
   
  </td>
  
  
  
   
   <td>        <div id="MSearchBox" class="MSearchBoxInactive">
        <span class="left">
          <img id="MSearchSelect" src="search/mag_sel.png"
               onmouseover="return searchBox.OnSearchSelectShow()"
               onmouseout="return searchBox.OnSearchSelectHide()"
               alt=""/>
          <input type="text" id="MSearchField" value="Search" accesskey="S"
               onfocus="searchBox.OnSearchFieldFocus(true)" 
               onblur="searchBox.OnSearchFieldFocus(false)" 
               onkeyup="searchBox.OnSearchFieldChange(event)"/>
          </span><span class="right">
            <a id="MSearchClose" href="javascript:searchBox.CloseResultsWindow()"><img id="MSearchCloseImg" border="0" src="search/close.png" alt=""/></a>
          </span>
        </div>
</td>
   
  
 </tr>
 </tbody>
</table>
</div>

<!-- Generated by Doxygen 1.7.6.1 -->
<script type="text/javascript">
var searchBox = new SearchBox("searchBox", "search",false,'Search');
</script>
</div>
<div id="side-nav" class="ui-resizable side-nav-resizable">
  <div id="nav-tree">
    <div id="nav-tree-contents">
    </div>
  </div>
  <div id="splitbar" style="-moz-user-select:none;" 
       class="ui-resizable-handle">
  </div>
</div>
<script type="text/javascript">
  initNavTree('extract__gis__data_8py.html','');
</script>
<div id="doc-content">
<div class="header">
  <div class="headertitle">
<div class="title">extract_gis_data.py</div>  </div>
</div><!--header-->
<div class="contents">
<a href="extract__gis__data_8py.html">Go to the documentation of this file.</a><div class="fragment"><pre class="fragment"><a name="l00001"></a><a class="code" href="namespaceambhas_1_1extract__gis__data.html">00001</a> <span class="comment"># -*- coding: utf-8 -*-</span>
<a name="l00002"></a>00002 <span class="stringliteral">&quot;&quot;&quot;</span>
<a name="l00003"></a>00003 <span class="stringliteral">Created on Mon Feb 27 19:18:08 2012</span>
<a name="l00004"></a>00004 <span class="stringliteral"></span>
<a name="l00005"></a>00005 <span class="stringliteral">@author: Sat Kumar Tomer</span>
<a name="l00006"></a>00006 <span class="stringliteral">@website: www.ambhas.com</span>
<a name="l00007"></a>00007 <span class="stringliteral">@email: satkumartomer@gmail.com</span>
<a name="l00008"></a>00008 <span class="stringliteral">&quot;&quot;&quot;</span>
<a name="l00009"></a>00009 
<a name="l00010"></a>00010 <span class="comment"># import required libraries</span>
<a name="l00011"></a>00011 <span class="keyword">from</span> osgeo.gdalconst <span class="keyword">import</span> *
<a name="l00012"></a>00012 <span class="keyword">import</span> gdal, xlrd,xlwt
<a name="l00013"></a>00013 <span class="keyword">import</span> numpy <span class="keyword">as</span> np
<a name="l00014"></a>00014 <span class="keyword">from</span> ambhas.gis <span class="keyword">import</span> utm2image
<a name="l00015"></a>00015 
<a name="l00016"></a><a class="code" href="namespaceambhas_1_1extract__gis__data.html#a3f0f14ae283914fb6d9c34915c288ad6">00016</a> <span class="keyword">def </span><a class="code" href="namespaceambhas_1_1extract__gis__data.html#a3f0f14ae283914fb6d9c34915c288ad6">extract_gis</a>(xls_in, xls_out, ds, ds_short_name):
<a name="l00017"></a>00017     <span class="stringliteral">&quot;&quot;&quot;</span>
<a name="l00018"></a>00018 <span class="stringliteral">    it reads the gis file defined in the ds</span>
<a name="l00019"></a>00019 <span class="stringliteral">    then extract the data at coordinates defined in each sheet of the xls_in file</span>
<a name="l00020"></a>00020 <span class="stringliteral">    and then write the data in the xls_out file</span>
<a name="l00021"></a>00021 <span class="stringliteral">    the header of the data in the xls_out are written as defined in the </span>
<a name="l00022"></a>00022 <span class="stringliteral">    ds_short_name</span>
<a name="l00023"></a>00023 <span class="stringliteral">    </span>
<a name="l00024"></a>00024 <span class="stringliteral">    xls_in: the name of the input xls file containing the co-ordinates of the plot</span>
<a name="l00025"></a>00025 <span class="stringliteral">    xls_out: the xls file in which the output will be written</span>
<a name="l00026"></a>00026 <span class="stringliteral">    ds: the data source file name in the gis format, these files must be in the </span>
<a name="l00027"></a>00027 <span class="stringliteral">        tiff format</span>
<a name="l00028"></a>00028 <span class="stringliteral">    ds_short_name:  the name that will appear as header in the output xls file</span>
<a name="l00029"></a>00029 <span class="stringliteral">    &quot;&quot;&quot;</span>
<a name="l00030"></a>00030     
<a name="l00031"></a>00031     <span class="keywordflow">if</span> type(ds) <span class="keywordflow">is</span> <span class="keywordflow">not</span> list:
<a name="l00032"></a>00032         <span class="keywordflow">raise</span> TypeError(<span class="stringliteral">&#39;input ds should be of list type&#39;</span>)
<a name="l00033"></a>00033     
<a name="l00034"></a>00034     book_out = xlwt.Workbook()
<a name="l00035"></a>00035 
<a name="l00036"></a>00036     final_data = np.empty((66,2,len(ds)))
<a name="l00037"></a>00037     <span class="keywordflow">for</span> k <span class="keywordflow">in</span> range(len(ds)):
<a name="l00038"></a>00038         dataset = gdal.Open(ds[k], GA_ReadOnly)
<a name="l00039"></a>00039         data = dataset.GetRasterBand(1).ReadAsArray() 
<a name="l00040"></a>00040         GT = dataset.GetGeoTransform()
<a name="l00041"></a>00041         
<a name="l00042"></a>00042         book = xlrd.open_workbook(xls_in)
<a name="l00043"></a>00043         
<a name="l00044"></a>00044         <span class="keywordflow">for</span> i <span class="keywordflow">in</span> xrange(66):
<a name="l00045"></a>00045             sheet = book.sheet_by_name(str(i+1))
<a name="l00046"></a>00046             xy = np.empty((sheet.nrows-1,2))
<a name="l00047"></a>00047             <span class="keywordflow">for</span> j <span class="keywordflow">in</span> range(xy.shape[0]):
<a name="l00048"></a>00048                 xy[j,0] = sheet.cell_value(j+1,0)    
<a name="l00049"></a>00049                 xy[j,1] = sheet.cell_value(j+1,1)
<a name="l00050"></a>00050                     
<a name="l00051"></a>00051             x,y = <a class="code" href="namespaceambhas_1_1gis.html#ac08652456f1fda96c33b457ff3e22fcf">utm2image</a>(GT,xy)
<a name="l00052"></a>00052                 
<a name="l00053"></a>00053             extracted_data = data[y,x]
<a name="l00054"></a>00054             final_data[i,0,k] = np.median(extracted_data)
<a name="l00055"></a>00055             final_data[i,1,k] = np.std(extracted_data)
<a name="l00056"></a>00056     
<a name="l00057"></a>00057     dataset = <span class="keywordtype">None</span>
<a name="l00058"></a>00058     
<a name="l00059"></a>00059     sheet_median = book_out.add_sheet(<span class="stringliteral">&#39;median&#39;</span>)   
<a name="l00060"></a>00060     sheet_std = book_out.add_sheet(<span class="stringliteral">&#39;std&#39;</span>)
<a name="l00061"></a>00061     
<a name="l00062"></a>00062     sheet_median.write(0,0,<span class="stringliteral">&#39;Plot no.&#39;</span>)
<a name="l00063"></a>00063     sheet_std.write(0,0,<span class="stringliteral">&#39;Plot no.&#39;</span>)
<a name="l00064"></a>00064     
<a name="l00065"></a>00065     <span class="keywordflow">for</span> i <span class="keywordflow">in</span> range(final_data.shape[0]):
<a name="l00066"></a>00066         sheet_median.write(i+1,0,i+1)
<a name="l00067"></a>00067         sheet_std.write(i+1,0,i+1)
<a name="l00068"></a>00068     
<a name="l00069"></a>00069     <span class="keywordflow">for</span> i <span class="keywordflow">in</span> range(len(ds)):
<a name="l00070"></a>00070         sheet_median.write(0,i+1,ds_short_name[i])
<a name="l00071"></a>00071         sheet_std.write(0,i+1,ds_short_name[i])
<a name="l00072"></a>00072     
<a name="l00073"></a>00073     <span class="keywordflow">for</span> i <span class="keywordflow">in</span> range(final_data.shape[0]):
<a name="l00074"></a>00074         <span class="keywordflow">for</span> j <span class="keywordflow">in</span> range(final_data.shape[2]):
<a name="l00075"></a>00075             sheet_median.write(i+1, j+1, final_data[i,0,j])
<a name="l00076"></a>00076             sheet_std.write(i+1, j+1, final_data[i,1,j])
<a name="l00077"></a>00077     book_out.save(xls_out)
</pre></div></div><!-- contents -->
</div>
<!-- window showing the filter options -->
<div id="MSearchSelectWindow"
     onmouseover="return searchBox.OnSearchSelectShow()"
     onmouseout="return searchBox.OnSearchSelectHide()"
     onkeydown="return searchBox.OnSearchSelectKey(event)">
<a class="SelectItem" href="javascript:void(0)" onclick="searchBox.OnSelectItem(0)"><span class="SelectionMark">&#160;</span>All</a><a class="SelectItem" href="javascript:void(0)" onclick="searchBox.OnSelectItem(1)"><span class="SelectionMark">&#160;</span>Classes</a><a class="SelectItem" href="javascript:void(0)" onclick="searchBox.OnSelectItem(2)"><span class="SelectionMark">&#160;</span>Namespaces</a><a class="SelectItem" href="javascript:void(0)" onclick="searchBox.OnSelectItem(3)"><span class="SelectionMark">&#160;</span>Files</a><a class="SelectItem" href="javascript:void(0)" onclick="searchBox.OnSelectItem(4)"><span class="SelectionMark">&#160;</span>Functions</a><a class="SelectItem" href="javascript:void(0)" onclick="searchBox.OnSelectItem(5)"><span class="SelectionMark">&#160;</span>Variables</a></div>

<!-- iframe showing the search results (closed by default) -->
<div id="MSearchResultsWindow">
<iframe src="javascript:void(0)" frameborder="0" 
        name="MSearchResults" id="MSearchResults">
</iframe>
</div>

  <div id="nav-path" class="navpath">
    <ul>
      <li class="navelem"><a class="el" href="extract__gis__data_8py.html">extract_gis_data.py</a>      </li>

    <li class="footer">Generated on Sat Jul 21 2012 12:26:08 for AMBHAS by
    <a href="http://www.doxygen.org/index.html">
    <img class="footer" src="doxygen.png" alt="doxygen"/></a> 1.7.6.1 </li>
   </ul>
 </div>


</body>
</html>
